import os

from langchain_openai import AzureChatOpenAI

os.environ["AZURE_OPENAI_ENDPOINT"] = "http://menshen.test.xdf.cn"
os.environ["OPENAI_API_KEY"] = "c8575027653b42b1b47747f0b4ab135b"
os.environ["OPENAI_API_TYPE"] = "azure"
os.environ["OPENAI_API_VERSION"] = "2023-05-15"


def get_llm():
    llm = AzureChatOpenAI(
        deployment_name="gpt-4o",
        model_name="gpt-4o",
        temperature=0)
    return llm
